package cn.ddm.hrm.service.impl;

import cn.ddm.hrm.domain.CourseType;
import cn.ddm.hrm.mapper.CourseTypeMapper;
import cn.ddm.hrm.service.ICourseTypeService;
import cn.ddm.hrm.vo.CrumbsVo;
import com.baomidou.mybatisplus.mapper.EntityWrapper;
import com.baomidou.mybatisplus.service.impl.ServiceImpl;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.cache.annotation.Cacheable;
import org.springframework.data.redis.core.RedisTemplate;
import org.springframework.stereotype.Service;

import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;

/**
 * <p>
 * 课程目录 服务实现类
 * </p>
 *
 * @author ldm
 * @since 2022-01-03
 */
//@Service
public class CourseTypeServiceImpl_001 extends ServiceImpl<CourseTypeMapper, CourseType> implements ICourseTypeService {

    @Autowired
    private RedisTemplate redisTemplate;

    private static final String COURSE_CACHE = "COURSE_CACHE";

    @Autowired
    private CourseTypeMapper courseTypeMapper;

    @Override
    @Cacheable(cacheNames = COURSE_CACHE, key = "'ALL'")
    public List<CourseType> treeData(long pid) {
        // 1.递归思想
//        return loadTreeData(pid);

        // 2.循环
        return loadTreeDataByLoop(pid);

        // 3.SQL递归
//        return courseTypeMapper.loadTreeDataBySql(pid);
    }

    @Override
    public List<CrumbsVo> crumbs(Long typeId) {
        return null;
    }

    /**
     * 循环
     * @param pid
     * @return
     */
    private List<CourseType> loadTreeDataByLoop(Long pid) {
        // 1.结果集
        List<CourseType> result = new ArrayList<>();

        // 2.查询所有节点
        List<CourseType> courseTypes = courseTypeMapper.selectList(null);

        // 用一个map装进
        HashMap<Long, CourseType> map = new HashMap<>();
        for (CourseType courseType : courseTypes) {
            map.put(courseType.getId(), courseType);
        }
        // 遍历,判断是否为当前节点
        courseTypes.forEach(courseType -> {
            Long pid1 = courseType.getPid();
            if (pid.longValue() == pid1.longValue())
                result.add(courseType);
            else {
                // 获取父亲
                CourseType parent  = map.get(pid1);
                // 放进去
                parent.getChildren().add(courseType);
            }
        });

        // 返回结果
        return result;

    }

    /**
     * 递归查询
     * @param pid
     * @return
     */
    private List<CourseType> loadTreeData(long pid) {
        List<CourseType> courseTypes = courseTypeMapper.selectList(new EntityWrapper<CourseType>().eq("pid", pid));
        // 出口

        if (courseTypes == null)
            return null;

        courseTypes.forEach(courseType -> {
            Long id = courseType.getId();
            List<CourseType> courseTypeList = loadTreeData(id);
            courseType.setChildren(courseTypeList);
        });

        return courseTypes;

    }


}
